
export enum Page {
  Train = '训练模型',
  Validate = '验证模型',
  Test = '测试模型',
  UserTest = '用户测试',
  GenerateJson = '生成JSON',
}

export enum TaskType {
  Training = '训练',
  Validation = '验证',
  Testing = '测试',
}

export interface ChartDataPoint {
  step: number;
  value: number;
}

export interface ModelInfo {
  id: string;
  name: string;
  createdAt: string;
}

export interface TrainingParameters {
  batchSize: number;
  numEpochs: number;
  learningRate: number;
  momentum: number;
  numWorkers: number;
}

export interface PredictionResult {
    label: string;
    probability: number;
}
